Edge detection Python